#include <muduo/net/TcpServer.h>
#include <muduo/net/EventLoop.h>
#include <iostream>
#include <functional>
using namespace std;
using namespace muduo;
using namespace muduo::net;
using namespace placeholders;
/*
1.组合tcpserver对象
2.创建EventLoop事件循环对象的指针
3.明确TCPserver狗仔函数需要什么参数，输出ChatServer的构造函数
4.在当前服务器类的构造函数当中，注册处理连接的回调函数和处理读写时间的回调函数
5.设置合适的服务端线程数量，muduo库会自己分配I/O线程和worker线程
*/
class ChatServer
{
public:
    ChatServer(EventLoop* loop, //事件循环
            const InetAddress& listenAddr,  //IP+Port
            const string& nameArg)
            :_server(loop,listenAddr,nameArg),_loop(loop)
            {
                //给服务器注册用户连接的创建和断开回调
                _server.setConnectionCallback(std::bind(&ChatServer::onConnection,this,_1));    //最后一位占位符
                //给服务器注册用户读写事件回调
                _server.setMessageCallback(std::bind(&ChatServer::onMessage,this,_1,_2,_3));

                //设置服务器的线程数量 1个I/O线程，3个worker线程
                _server.setThreadNum(4);
            }

            //开启事件循环
            void start(){
                _server.start();
            }
private:
    //专门处理用户的连接创建和断开 epoll listenfd accept
    void onConnection(const TcpConnectionPtr    &conn){
        if(conn->connected())
        {
            cout<<conn->peerAddress().toIpPort()<<" -> "<<
            conn->localAddress().toIpPort()<<" state:online"<< endl;
        }
        else
        {
            cout<<conn->peerAddress().toIpPort()<<" -> "<<
            conn->localAddress().toIpPort()<<" state:offline"<< endl;
            conn->shutdown(); // close(fd)
            //_loop->quit();    //退出epoll
        }

    }
    //专门处理用户的读写事件
    void onMessage(const TcpConnectionPtr &conn,    //连接
                    Buffer *buffer,    //缓冲区
                    Timestamp time)     //接收到数据的时间信息
    {
        string buf=buffer->retrieveAllAsString();
        cout<<"recv data:"<<buf<<" time:"<< time.toString()<<endl;
        conn->send(buf);
    }
    TcpServer _server;//#1
    EventLoop *_loop;//#2
};

int main()  //开服务器后可以用Linux命令telnet ip port来测试通信是否畅通
{
    EventLoop loop;//epoll
    InetAddress addr("127.0.0.1",6000);
    ChatServer server(&loop,addr,"ChatServer");

    server.start();
    loop.loop();    //epoll_wait以阻塞方式等待新用户连接，已连接用户的读写事件等

    return 0;
}